Order-Only On-Policy Distillation
序数在策略蒸馏:只学教师 Top-K 排序
🧭 LLM 知识蒸馏 · 序数监督
University of Memphis · University of Missouri
O3PD 在策略蒸馏中只复现教师 top-K token 排序,不拟合概率值,对温度不变并在数学推理上全面优于 OPD
📌 工作参考价值(4 点)
- 在蒸馏 LLM 到线上小模型时,不要强制小模型拟合教师 soft probabilities 的绝对数值,尤其当教师温度需要搜索时;改用 top-K 排序监督(如 ListMLE、RankNet)可省去温度调参,训练更稳定。
- 生成式推荐/query 生成的 on-policy 蒸馏中,教师概率值混杂其自身校准与置信度,小容量学生难以复现;只保留序数信息能降低容量要求,让模型专注学习候选 token 的相对好坏,适合电商 query 推荐、广告文案生成、推荐理由生成等场景。
- 若已有 OPD 流水线,可将 loss 替换为对 teacher top-K logits 的 ranking loss,保留 on-policy 生成模式;数学推理任务上的 +2.9 平均分提升说明该 trick 在复杂生成任务上可能有效。
- 对 Agent 策略蒸馏的启示:当学生策略模型容量有限时,教师动作分布的序数偏好可能比概率值更重要,可以在策略优化中借鉴序数监督。